import requests
import json

# 定义API的URL
api_url = "http://81.70.84.40:8000/v1/chat/completions"  # 替换为实际的API服务器地址

token = "eyJhbGciOiJIUzUxMiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJ1c2VyLWNlbnRlciIsImV4cCI6MTczMzQ3NDQ3NiwiaWF0IjoxNzI1Njk4NDc2LCJqdGkiOiJjcmUxM2I2Ymk3c2V0bDM5cTExMCIsInR5cCI6InJlZnJlc2giLCJzdWIiOiJjcW1ndXZ1Y3A3ZjFocmY4Y24xMCIsInNwYWNlX2lkIjoiY3FtZ3V2dWNwN2YxaHJmOGNuMGciLCJhYnN0cmFjdF91c2VyX2lkIjoiY3FtZ3V2dWNwN2YxaHJmOGNuMDAifQ.1RCp2YGziSbI6Yd3IwxhNrbPNT1Up4mFHRvPlb9qoC-azw3EPUk8WRmMBBFCc_6DC5twr0Ea6oE0SwP59Ok19g"
# 定义Authorization Token
headers = {
    "Authorization": f"Bearer {token}",  # 替换为你的实际Token
    "Content-Type": "application/json"
}

# pdf_url = "https://aclanthology.org/2024.findings-acl.944.pdf"
pdf_url = "https://arxiv.org/pdf/2409.03271"
text = """
请阅读这篇论文，然后依次完成下面的工作：
1. 请使用3-5个要点说明本文的主要创新点。如果创新点较多，可以使用更多的要点。
2. 除开主要创新点外，这篇论文是否提供或说明了其他重要的要点。包括但不限于：实验的深入分析、方法或算法的不足和限制、可供发展或应用推广的角度等。
3. 本文提出的算法或方法，是否是通过观察到某些模型行为或特征而引出的。比如，因观察模型表现出某些现象而提出了具体的改进方法。如果存在这种情况，请你说明被观察到的现象是什么？现象产生了怎样的影响？
"""

# 构建请求数据
data = {
    "model": "kimi",  # 可以随意填写，或使用特定模型ID
    "messages": [
        {
            "role": "user",
            "content": [
                {
                    "type": "file",
                    "file_url": {
                        "url": pdf_url  # 文件URL
                    }
                },
                {
                    "type": "text",
                    "text": text  # 用户问题
                }
            ]
        }
    ],
    "use_search": False  # 关闭联网搜索
}

# 将数据转换为JSON格式
json_data = json.dumps(data)

# 发送POST请求
response = requests.post(api_url, headers=headers, data=json_data)

# 解析响应数据
if response.status_code == 200:
    result = response.json()
    print("Response ID:", result.get("id"))
    print("Assistant Message:", result["choices"][0]["message"]["content"])
else:
    print(f"Request failed with status code {response.status_code}")
    print("Response:", response.text)

